package main

import (
	"context"
	"flag"
	"fmt"
	"go.uber.org/zap"
	"os"
	"os/signal"
	"proxy/conf"
	redisDB "proxy/dao"
	"proxy/logger"
	"proxy/tcp"
	"syscall"
	"time"
)

func main() {
	var filepath string
	flag.StringVar(&filepath, "filepath", "./conf/config.yaml", "configfilepath")
	flag.Parse()
	err := conf.Init(filepath)
	if err != nil {
		fmt.Println(err)
		return
	}
	if err := log.Init(conf.Conf.Mode); err != nil {
		fmt.Printf("init logger failed,err:=%v\n", err)
		return
	}
	ctx := context.Background()
	log.Info(ctx, "init log ok")
	if err := redisDB.Init(conf.Conf.RedisConfig); err != nil {
		fmt.Printf("init redis failed,err:=%v\n", err)
		return
	}
	if conf.Conf.MachineID == 1 {
		go tcp.ClientMain(conf.Conf.ServerPortConfg.LocaServerPort, conf.Conf.ServerPortConfg.Host)
	}
	if conf.Conf.MachineID == 2 {
		go tcp.ServerMain(conf.Conf.ServerPortConfg.EscServerPort)
	}
	quit := make(chan os.Signal, 1) // 创建一个接收信号的通道
	// kill 默认会发送 syscall.SIGTERM 信号
	// kill -2 发送 syscall.SIGINT 信号，我们常用的Ctrl+C就是触发系统SIGINT信号
	// kill -9 发送 syscall.SIGKILL 信号，但是不能被捕获，所以不需要添加它
	// signal.Notify把收到的 syscall.SIGINT或syscall.SIGTERM 信号转发给quit
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此处不会阻塞
	<-quit                                               // 阻塞在此，当接收到上述两种信号时才会往下执行
	zap.L().Info("Shutdown Server ...")
	// 创建一个5秒超时的context
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	// 5秒内优雅关闭服务（将未处理完的请求处理完再关闭服务），超过5秒就超时退出
	zap.L().Info("Server exiting")
}
